﻿#region Copyright

// The contents of this file are subject to the Mozilla Public License
//  Version 1.1 (the "License"); you may not use this file except in compliance
//  with the License. You may obtain a copy of the License at
//  
//  http://www.mozilla.org/MPL/
//  
//  Software distributed under the License is distributed on an "AS IS"
//  basis, WITHOUT WARRANTY OF ANY KIND, either express or implied. See the
//  License for the specific language governing rights and limitations under 
//  the License.
//  
//  The Initial Developer of the Original Code is Robert Smyth.
//  Portions created by Robert Smyth are Copyright (C) 2008.
//  
//  All Rights Reserved.

#endregion

using System.Collections;
using System.Collections.Generic;


namespace NUnitGrid.Server.Plugins.Nodes
{
    public class OnlineNodesCollection : IEnumerable<IGridNode>
    {
        private readonly List<IGridNode> onlineNodes = new List<IGridNode>();

        public int Count
        {
            get { return onlineNodes.Count; }
        }

        public int BusyNodesCount
        {
            get
            {
                var busyNodesCount = 0;
                foreach (var node in onlineNodes)
                {
                    if (node.Busy)
                    {
                        busyNodesCount++;
                    }
                }
                return busyNodesCount;
            }
        }

        public IEnumerator<IGridNode> GetEnumerator()
        {
            return onlineNodes.GetEnumerator();
        }

        IEnumerator IEnumerable.GetEnumerator()
        {
            return GetEnumerator();
        }

        public bool Contains(IGridNode item)
        {
            return onlineNodes.Contains(item);
        }

        public IGridNode[] ToArray()
        {
            return onlineNodes.ToArray();
        }

        public bool Remove(IGridNode item)
        {
            return onlineNodes.Remove(item);
        }

        public void Clear()
        {
            onlineNodes.Clear();
        }

        public void OnNodeOnline(IGridNode item)
        {
            onlineNodes.Add(item);
        }

        public void OnNodeOffline(IGridNode node)
        {
            if (onlineNodes.Contains(node))
            {
                onlineNodes.Remove(node);
            }
        }

        public IGridNode GetFirstAvailableNode()
        {
            IGridNode availableNode = null;
            foreach (var node in onlineNodes)
            {
                if (!node.Busy)
                {
                    availableNode = node;
                    break;
                }
            }
            return availableNode;
        }
    }
}